

package com.hazelcast.jet.impl.client.protocol.task;

import com.hazelcast.client.impl.protocol.ClientMessage;
import com.hazelcast.client.impl.protocol.codec.JetUpdateJobConfigCodec;
import com.hazelcast.client.impl.protocol.codec.JetUpdateJobConfigCodec.RequestParameters;
import com.hazelcast.instance.impl.Node;
import com.hazelcast.internal.nio.Connection;
import com.hazelcast.internal.serialization.Data;
import com.hazelcast.jet.config.DeltaJobConfig;
import com.hazelcast.jet.impl.operation.UpdateJobConfigOperation;
import com.hazelcast.security.permission.ActionConstants;
import com.hazelcast.spi.impl.operationservice.Operation;

import javax.annotation.Nullable;

public class JetUpdateJobConfigMessageTask extends AbstractJetMessageTask<RequestParameters, Data> {

    protected JetUpdateJobConfigMessageTask(ClientMessage clientMessage, Node node, Connection connection) {
        super(clientMessage, node, connection, JetUpdateJobConfigCodec::decodeRequest, JetUpdateJobConfigCodec::encodeResponse);
    }

    @Override
    protected Operation prepareOperation() {
        DeltaJobConfig deltaConfig = serializationService.toObject(parameters.deltaConfig);
        return new UpdateJobConfigOperation(parameters.jobId, deltaConfig);
    }

    @Override
    protected Object processResponseBeforeSending(Object response) {
        return toData(response);
    }

    @Override
    public String getMethodName() {
        return "updateConfig";
    }

    @Override
    public Object[] getParameters() {
        return new Object[0];
    }

    @Nullable
    @Override
    public String[] actions() {
        return new String[]{ActionConstants.ACTION_RESTART};
    }
}
